from plotly.graph_objs import Bar, Layout
from plotly import offline

from die import Die

die_1 = Die()
die_2 = Die(10)

results = [die_1.roll() + die_2.roll() for _ in range(50_000)]

max = die_1.num_sides + die_2.num_sides
frequencies = [results.count(val) for val in range(2, max + 1)]

x_values = list(range(2, max + 1))
data = [Bar(x=x_values, y=frequencies)]

x_axis_config = {'title': '结果', 'dtick': 1}
y_axis_config = {'title': '频率'}

my_layout = Layout(title='掷一个D6和D10 50 000次的结果',
                   xaxis=x_axis_config,
                   yaxis=y_axis_config)

offline.plot({'data': data, 'layout': my_layout}, filename='output/d6_d10.html')
